import PIL.Image as Image
import os

IMAGES_PATH = r'C:\Users\Admin\Desktop\unet-crack\display\out/'  # 图片集地址
IMAGES_FORMAT = ['.jpg', '.png']  # 图片格式
IMAGE_SIZE = [718,363]  # 每张小图片的大小
IMAGE_ROW = 12  # 图片间隔，也就是合并成一张图后，一共有几行
IMAGE_COLUMN = 10  # 图片间隔，也就是合并成一张图后，一共有几列
IMAGE_SAVE_PATH = r'G:\testout\555/'  # 图片转换后的地址

# 获取图片集地址下的所有图片名称
image_names = []
count=0
for i in range(1,IMAGE_ROW+1):
    for j in range(IMAGE_COLUMN):
        count+=1
        image_names.append(str(count)+'.png')

# 简单的对于参数的设定和实际图片集的大小进行数量判断
if len(image_names) != IMAGE_ROW * IMAGE_COLUMN:
    raise ValueError("合成图片的参数和要求的数量不能匹配！")


# 定义图像拼接函数
def image_compose():
    to_image = Image.new('RGB', (IMAGE_COLUMN * IMAGE_SIZE[0], IMAGE_ROW * IMAGE_SIZE[1]))  # 创建一个新图
    # 循环遍历，把每张图片按顺序粘贴到对应位置上
    for y in range(IMAGE_ROW):
        for x in range(IMAGE_COLUMN):
            from_image = Image.open(IMAGES_PATH + image_names[IMAGE_COLUMN * (y - 1) + x - 1]).resize(
                (IMAGE_SIZE[0], IMAGE_SIZE[1]), Image.ANTIALIAS)
            to_image.paste(from_image, ((x - 1) * IMAGE_SIZE[0], (y - 1) * IMAGE_SIZE[1]))
    return to_image.save(IMAGE_SAVE_PATH+'123555.png')  # 保存新图


image_compose()  # 调用函数